
/**
 * OpenEmulator
 * Mac OS X Chooser View Controller
 * (C) 2009-2010 by Marc S. Ressl (mressl@umich.edu)
 * Released under the GPL
 *
 * Controls a template chooser view
 */

#import "Quartz/Quartz.h"

#import "TemplateChooserViewController.h"
#import "TemplateChooserItem.h"
#import "Document.h"

#define USER_TEMPLATES_GROUP @"My Templates"
#define SPLIT_VERT_MIN 128
#define SPLIT_VERT_MAX 256
#define SPLIT_HORIZ_MIN 108

#define EMULATION_PACKAGE_PATH_EXTENSION	@"emulation"
#define EMULATION_FILE_PATH_EXTENSION		@"xml"

@implementation TemplateChooserViewController

- (id)init
{
    self = [super initWithNibName:@"TemplateChooserView" bundle:nil];
    
    if (self)
    {
        groups = [[NSMutableArray alloc] init];
        items = [[NSMutableDictionary alloc] init];
    }
    
    return self;
}

- (void)dealloc
{
    [groups release];
    [items release];
    
    [selectedGroup release];
    
    [super dealloc];
}

- (void)setDelegate:(id)theDelegate
{
    delegate = theDelegate;
}

- (void)reloadData
{
    NSString *group = [self selectedGroup];
    NSString *path = [self selectedItemPath];
    
    [self loadGroups];
    
    [fTableView reloadData];
    [fImageBrowserView reloadData];
    
    [self selectGroup:group andItemWithPath:path];
}

- (void)awakeFromNib
{
    [self loadGroups];
    
    tableCell = [[VerticallyCenteredTextFieldCell alloc] init];
    [tableCell setControlSize:NSSmallControlSize];
    [tableCell setFont:[NSFont labelFontOfSize:[NSFont smallSystemFontSize]]];
    [tableCell setHorizontalInset:5.0];
    
    NSDictionary *titleAttributes;
    titleAttributes = [NSDictionary dictionaryWithObjectsAndKeys:
                       [NSFont messageFontOfSize:11.0f], NSFontAttributeName,
                       [NSColor blackColor], NSForegroundColorAttributeName,
                       nil];
    NSDictionary *subtitleAttributes;
    subtitleAttributes = [NSDictionary dictionaryWithObjectsAndKeys:
                          [NSFont messageFontOfSize:11.0f], NSFontAttributeName,
                          [NSColor whiteColor], NSForegroundColorAttributeName,
                          nil];
    [fImageBrowserView setAllowsEmptySelection:NO];
    [fImageBrowserView setAllowsMultipleSelection:NO];
    [fImageBrowserView setCellsStyleMask:IKCellsStyleTitled];
    [fImageBrowserView setCellSize:NSMakeSize(108, 64)];
    [fImageBrowserView setDelegate:self];
    [fImageBrowserView setDataSource:self];
    [fImageBrowserView setValue:titleAttributes
                         forKey:IKImageBrowserCellsTitleAttributesKey];
    [fImageBrowserView setValue:subtitleAttributes
                         forKey:IKImageBrowserCellsHighlightedTitleAttributesKey];
    
    [fTableView setDelegate:self];
    [fTableView setDataSource:self];
}



- (BOOL)templatesAtPathValid:(NSString *)groupPath
{
    groupPath = [groupPath stringByExpandingTildeInPath];
    NSFileManager *fileManager = [NSFileManager defaultManager];
    BOOL isDirectory;
    if (![fileManager fileExistsAtPath:groupPath isDirectory:&isDirectory] ||
        !isDirectory)
        return NO;
    
    NSArray *subpaths = [fileManager contentsOfDirectoryAtPath:groupPath
                                                         error:nil];
    for (NSString *pathComponent in subpaths)
    {
        NSString *pathExtension = [[pathComponent pathExtension] lowercaseString];
        
        if (([pathExtension compare:EMULATION_PACKAGE_PATH_EXTENSION] == NSOrderedSame) ||
            ([pathExtension compare:EMULATION_FILE_PATH_EXTENSION] == NSOrderedSame))
            return YES;
    }
    
    return NO;
}

- (void)addTemplatesAtPath:(NSString *)thePath
                   toGroup:(NSString *)theGroup
{
    if (![items objectForKey:theGroup])
        [items setObject:[NSMutableArray array] forKey:theGroup];
    
    thePath = [thePath stringByExpandingTildeInPath];
    
    NSArray *pathContents = [[NSFileManager defaultManager]
                             contentsOfDirectoryAtPath:thePath
                             error:nil];
    for (NSString *filename in pathContents)
    {
        NSString *path = [thePath stringByAppendingPathComponent:filename];
        
        NSString *pathExtension = [[path pathExtension] lowercaseString];
        if (([pathExtension compare:EMULATION_PACKAGE_PATH_EXTENSION] != NSOrderedSame) &&
            ([pathExtension compare:EMULATION_FILE_PATH_EXTENSION] != NSOrderedSame))
            continue;
        
        TemplateChooserItem *item;
        item = [[TemplateChooserItem alloc] initWithOEDocumentPath:path];
        
        if (item)
            [[items objectForKey:theGroup] addObject:item];
    }
}

- (void)loadGroups
{
    NSString *templatesPath = [[[NSBundle mainBundle] resourcePath]
                               stringByAppendingPathComponent:@"templates"];
    
    [groups removeAllObjects];
    
    // Find templates
    NSFileManager *fileManager = [NSFileManager defaultManager];
    NSArray *subpaths = [fileManager contentsOfDirectoryAtPath:templatesPath
                                                         error:nil];
    for (NSString *pathComponent in subpaths)
    {
        NSString *groupPath = [templatesPath stringByAppendingPathComponent:pathComponent];
        if ([self templatesAtPathValid:groupPath])
            [groups addObject:pathComponent];
    }
    
    // Sort alphabetically
    [groups setArray:[groups sortedArrayUsingSelector:@selector(compare:)]];
    
    // Find user templates
    [items removeObjectForKey:NSLocalizedString(USER_TEMPLATES_GROUP,
                                                "Template Chooser.")];
    if ([self templatesAtPathValid:USER_TEMPLATES_FOLDER])
        [groups addObject:NSLocalizedString(USER_TEMPLATES_GROUP,
                                            "Template Chooser.")];
}

- (void)loadItems
{
    if ([items objectForKey:selectedGroup])
        return;
    
    NSString *group = NSLocalizedString(USER_TEMPLATES_GROUP,
                                        "Template Chooser.");
    if ([selectedGroup compare:group] != NSOrderedSame)
    {
        NSString *templatesPath = [[[NSBundle mainBundle] resourcePath]
                                   stringByAppendingPathComponent:@"templates"];
        NSString *groupPath = [templatesPath stringByAppendingPathComponent:selectedGroup];
        [self addTemplatesAtPath:groupPath
                         toGroup:selectedGroup];
    }
    else
        [self addTemplatesAtPath:USER_TEMPLATES_FOLDER
                            toGroup:group];
}



- (void)splitView:(NSSplitView *)sender
resizeSubviewsWithOldSize:(NSSize)oldSize
{
    NSArray *subviews = [sender subviews];
    
    NSSize newSize = [sender frame].size;
    float deltaWidth = newSize.width - oldSize.width;
    float deltaHeight = newSize.height - oldSize.height;
    
    for (int i = 0; i < [subviews count]; i++)
    {
        NSView *subview = [subviews objectAtIndex:i];
        NSRect frame = subview.frame;
        
        if ([sender isVertical])
        {
            frame.size.height += deltaHeight;
            if (i == 1)
                frame.size.width += deltaWidth;
        }
        else
        {
            frame.size.width += deltaWidth;
            if (i == 0)
            {
                frame.size.height += deltaHeight;
                if (frame.size.height < SPLIT_HORIZ_MIN)
                    frame.size.height = SPLIT_HORIZ_MIN;
            }
            else
            {
                frame.origin.y += deltaHeight;
                float maxHeight = (newSize.height - [sender dividerThickness] -
                                   SPLIT_HORIZ_MIN);
                if (frame.size.height >= maxHeight)
                {
                    frame.origin.y += frame.size.height - maxHeight;
                    frame.size.height = maxHeight;
                }
            }
        }
        
        [subview setFrame:frame];
    }
}

- (CGFloat)splitView:(NSSplitView *)splitView
constrainMinCoordinate:(CGFloat)proposedMin
         ofSubviewAt:(NSInteger)dividerIndex
{
    if ((splitView == fVerticalSplitView) && (dividerIndex == 0))
        return SPLIT_VERT_MIN;
    else if ((splitView = fHorizontalSplitView) && (dividerIndex == 0))
        return SPLIT_HORIZ_MIN;
    
    return proposedMin;
}

- (CGFloat)splitView:(NSSplitView *)splitView
constrainMaxCoordinate:(CGFloat)proposedMax
         ofSubviewAt:(NSInteger)dividerIndex
{
    if ((splitView == fVerticalSplitView) && (dividerIndex == 0))
        return SPLIT_VERT_MAX;
    
    return proposedMax;
}



- (NSInteger)numberOfRowsInTableView:(NSTableView *)aTableView
{
    return [groups count];
}

- (id)tableView:(NSTableView *)aTableView
objectValueForTableColumn:(NSTableColumn *)aTableColumn
            row:(NSInteger)rowIndex
{
    return [groups objectAtIndex:rowIndex];
}

- (void)tableViewSelectionDidChange:(NSNotification *)aNotification
{
    if (selectedGroup)
    {
        [selectedGroup release];
        selectedGroup = nil;
    }
    NSInteger rowIndex = [fTableView selectedRow];
    if (rowIndex != -1)
        selectedGroup = [[groups objectAtIndex:rowIndex] copy];
    
    [fImageBrowserView reloadData];
    [fImageBrowserView setSelectionIndexes:[NSIndexSet indexSetWithIndex:0]
                      byExtendingSelection:NO];
}

- (NSCell *)tableView:(NSTableView *)tableView
dataCellForTableColumn:(NSTableColumn *)tableColumn
                  row:(NSInteger)row
{
    return tableCell;
}



- (NSUInteger)numberOfItemsInImageBrowser:(IKImageBrowserView *)aBrowser
{
    if (!selectedGroup)
        return 0;
    
    if (![items objectForKey:selectedGroup])
        [self loadItems];
    
    return [[items objectForKey:selectedGroup] count];
}

- (id)imageBrowser:(IKImageBrowserView *)aBrowser itemAtIndex:(NSUInteger)index
{
    if (!selectedGroup)
        return nil;
    
    return [[items objectForKey:selectedGroup] objectAtIndex:index];
}

- (void)imageBrowserSelectionDidChange:(IKImageBrowserView *)aBrowser
{
    NSString *label = @"";
    NSImage *image = nil;
    NSString *description = @"";
    
    NSUInteger index = [[fImageBrowserView selectionIndexes] firstIndex];
    if (index != NSNotFound)
    {
        TemplateChooserItem *item = [self imageBrowser:fImageBrowserView
                                   itemAtIndex:index];
        label = [item imageTitle];
        image = [item imageRepresentation];
        description = [item description];
    }
    [fSelectedItemLabelView setStringValue:label];
    [fSelectedItemImageView setImage:image];
    [fSelectedItemDescriptionView setStringValue:description];
    
    if ([delegate respondsToSelector:@selector(templateChooserSelectionDidChange:)])
        [delegate templateChooserSelectionDidChange:self];
}

- (void)imageBrowser:(IKImageBrowserView *)aBrowser
cellWasDoubleClickedAtIndex:(NSUInteger)index
{
    if ([delegate respondsToSelector:@selector(templateChooserItemWasDoubleClicked:)])
        [delegate templateChooserItemWasDoubleClicked:self];
}



- (void)selectGroup:(NSString *)group
    andItemWithPath:(NSString *)path
{
    if (![groups count])
    {
        [fTableView selectRowIndexes:[NSIndexSet indexSet]
                byExtendingSelection:NO];
        
        [delegate templateChooserSelectionDidChange:self];
        return;
    }
    
    NSInteger rowIndex = 0;
    if ([groups containsObject:group])
        rowIndex = [groups indexOfObject:group];
    [fTableView selectRowIndexes:[NSIndexSet indexSetWithIndex:rowIndex]
            byExtendingSelection:NO];
    
    NSInteger itemIndex = 0;
    NSArray *groupItems = [items objectForKey:group];
    for (int i = 0; i < [groupItems count]; i++)
    {
        TemplateChooserItem *item = [groupItems objectAtIndex:i];
        if ([[item path] compare:path] == NSOrderedSame)
            itemIndex = i;
    }
    
    [fImageBrowserView setSelectionIndexes:[NSIndexSet indexSetWithIndex:itemIndex]
                      byExtendingSelection:NO];
    NSRect itemRect = [fImageBrowserView itemFrameAtIndex:itemIndex];
    [(NSView *)fImageBrowserView scrollPoint:itemRect.origin];
}

- (NSString *)selectedGroup
{
    if (selectedGroup)
        return [[selectedGroup copy] autorelease];
    else
        return nil;
}

- (NSString *)selectedItemPath
{
    NSUInteger imageIndex = [[fImageBrowserView selectionIndexes] firstIndex];
    if (imageIndex == NSNotFound)
        return nil;
    
    TemplateChooserItem *item = [self imageBrowser:fImageBrowserView
                                       itemAtIndex:imageIndex];
    return [[[item path] copy] autorelease];
}

@end
